| Name | Version | Summary | date |
| linax |
0.1.0 |
State Space Models in Jax |
2025-10-28 12:49:04 |
| optimistix |
0.0.11 |
Nonlinear optimisation in JAX and Equinox. |
2025-10-17 20:53:28 |
| typinox |
0.3.0 |
Enhanced runtime type-checked JAX modules |
2025-10-15 05:25:53 |
| equinox |
0.13.2 |
Elegant easy-to-use neural networks in JAX. |
2025-10-09 10:15:39 |
| cryojax |
0.5.0 |
Cryo-EM image simulation and analysis powered by JAX |
2025-10-07 18:59:05 |
| atmodeller |
0.9.0 |
Volatile partitioning between a rocky interior and atmosphere |
2025-08-03 07:32:46 |
| jaxtyping |
0.2.38 |
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. |
2025-02-13 22:02:20 |
| diffrax |
0.6.2 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-12-24 16:50:08 |
| lineax |
0.0.7 |
Linear solvers in JAX and Equinox. |
2024-10-21 10:54:53 |
| seasons-cal |
1.0.2 |
None |
2024-10-15 15:21:15 |
| distreqx |
0.0.1 |
Lightweight JAX library of probability distributions and bijectors. |
2024-09-15 06:13:45 |